Create max_flow.py