add the traffic light code