Added a websocket client to the generator