This should work for the menu part. Not yet compiled
Changed main.cpp to work with new menu