Adding to LED code