Fixed while-loop for main menu