Added mypy support.
Added feedback.