ENH: add support for python 3.x