Added restart, better game over screen