Added comments to .py file