Add basic websocket setup