Update lights code from @TeamCanjica github