Added victoryLights function and tidied up