Add code for leds with buttons