Changes to __init__.py