Added while loops. Not sure how to handle infinite loops for the user though. https://youtu.be/c4T-OxchO9s