Update __init__.py

Fixed relative import on Python 3k