Updated websocket endpoint