changed code to have it work with all 4 LED