Add play again prompt and bug fixes