Added a main input loop, and a prompt to play again