Putting together the switch and LED code