Add game over screen and some effects

Include bug with fast switch states running/paused