Update ws.py