Adding some comments in main.cpp