Update traffic_light.rb