Update __init__.py

Use relative import for it to be compatible with python3.x.