Update main.cpp

update