Add comments to the main python file