Add some lightOn logic