Main.cpp update