Updated Main.cpp