Add comments to the websocket client implementation