whip up figures in fluid_demo.py