Update lesson3_states.py