Update WebsocketClient.py