Update get_colors.py