Please complete the required checklist items in the PR description: ❌ You must confirm: I have added Google-style docstrings to important functions and ensured function parameters are typed.