updates to rainbow display and correctly handling game restart