Updated websocket module