add new feature for zookeeper