Main.cpp fixed