Use type annotations everywhere

[ypeels]

https://mypy.readthedocs.io/en/stable/cheat_sheet_py3.html

cleanup, documentation

[loganbvh]

I have to think a bit more about whether this is actually something that would make the code more readable/maintainable. I am not sure yet.

[loganbvh]

Closing because it's probably not worth it...