Updated main.cpp