Update main.cpp

confirm