WebSocket integration