Updated the code to cover the LEDS