Update ws_script.py