Simplify the LED code