Made some visual improvements, added a playAgain function