Update colorize_states.py