Update main.cpp

Edit comments