Traffic Lights
Revert "Traffic Lights"

This reverts commit bf26676407802af4016649c04804e3b1ef7938ca.
Optimised the code.

Removed an unnecessary variable and cleanup a code error on another line