Updated __init__.py for new submodules