Update websocket_script.py