Update main.cpp

ok