Update zookeeper.py