Added websocket support to graph