Added code to loop and listen for input on menu