Adding game over screen and restart game