Update __init__.py

Python 3 support