modify zk.go add TODO