Update main.cpp

complete