Added a game over screen, but very buggy.