Split TrafficLight class into a separate file