modified main.cpp comments