Added support for Python 3