Added traffic light code