Update __init__.py

Make it working with Python3