Add light update code. Added hacky code to make program wait.