python version of z, for experimentation - View it on GitHub