P4: Add while loop to allow user to return to menu.