Adding comments to main.cpp