Integrate websocket support