Update main.cpp

1.1 Update