Update __init__.py

python 3 support