basic_colors = ("green", "blue", "red", "yellow")
for color in basket_1.balloon_colors:
if color not in basic_colors:
basket_1.balloon_colors.remove(color)
# Remove the colors that are not in basic_colors
# from basket_2
for color in basket_2.balloon_colors:
if color not in basic_colors:
basket_2.balloon_colors.remove(color)