Add code for LED display