new_colors = ("red", "blue", "green", "purple", "yellow")
for color in new_colors:
print(color)
baskets[0].balloon_colors.add(color)
baskets[1].balloon_colors.add(color)