Added close option after game over