Add python types

[ilan-gold]

We can do this incrementally, adding mypy checks as well as we go if memory serves to make things more bite-sized