Update __init__.py for module