updating main.cpp