Added game over screen and ability to restart game