Create wm_plot.py