Switch to zlib

Easier to be compatible between python 2 and 3