Updating main.cpp