Update __init__.py

Fix import not working in Python 3.3