Add annotation type for all methods and classes and use mypy to check the annotation types.
Add annotation type for all methods and classes and use mypy to check the annotation types.