Code Refactor: Modified structure only (no code changes beyond added __init__.py and import statements).