Update __init__.py

Maintain backwards compat as much as is possible.