Add websocket client python