Update websocket.py