Add support for python3