Update __init__.py

for compat python 3 and 2