Add comments to the python file