Adds websocket setup