Cleaning up LED program a bit