Add flow plotting script