Update Flow.py function signature for run on save