Update websocket.js