Add wrapper for websocket.