Update python client for websocket