Add parameter to TrafficLight