Add restart option to game over screen