updated Main.cpp