Logic change to Lights On puzzle