Traffic Light Code Python Game