Code for traffic light project