One problem you might run into, is if the user enters something completely unexpected, like "rgfqerg", it'll quit, whereas it should tell the user it's bad input and ask for a yes or no answer again.