[Epic][Improvement] Improve strictness of code quality

[JamesKunstle]

We can make development easier on ourselves by adding a few requirements to our codebase:

1. consistently typing things and requiring static type checking like mypy
2. adding docstrings to all non-trivial functions