Update main.cpp

1