Added the websocket client class