Adding connection timeout to zookeeper