Add plot_flow.py file lost during merge