In order to be more developer friendly, the code should use type hint as much as possible. This would be useful for auto-completion to work within editors. It might also be a good idea to use Pydantic.