Update main.cpp
update