Add support for websocket