Add comments to main