Update __init__.py

Make it work for Python 3.x