Update main.cpp

edit comments