Adds support for Python 3