Add support for python 3