Final code version: addded lights