final code for traffic light