Update wm_func.py