Update main.cpp

Same here ...