Changed the way the play again button handlesPlay