Added restart after game over