Added the menu while loop on main.cpp