Update wm_main.py