Added function to restart game after loss instead of closing program