Type hints/mypy checks

[yschroeder]

Would it be useful to add type hints to the code?

I would be willing to contribute and maybe run some mypy checks (locally) to catch some bugs...