Update main.cpp

Added comments