Add traffic light test