Update main.cpp

add some comments