Update Sort_colors.py