[always] Type Hints and Documentation

[san-ghun]

While there are some type hints, they could be more comprehensive.

Consider:

• Adding return type hints to all functions
• Using more specific types instead of Any
• Adding comprehensive docstrings with examples