Change definition of Z_n

This commit changes Z_n, which I *think* resolves Referee 2s concern
in point 3.
Elaborate on O(1) term to satisfy referee