updated main.cpp comments