improving the refresh after game over