Add game over and restart capability.