Add comments to ZK mock
Refactor node creation stubbing in ZK mock