Updated to include websocket-client