add zktest code