Update __init__.py

Reverting changes here, moved them to master.