Add optional dependency for zookeeper