polish and game over stuff