Add support for Python 3