Update main.cpp
Update main.cpp