Fix __init__.py imports for python3