add support for python 3