Added code for lights
Solved using powers of 2