Added comments to flow.py