Update __init__.py for python 3