Updates to ZK schema