Added functionality to allow user to play again after game ends