Add mypy type signatures

Following up on !8 (closed) again, this adds the mypy type signatures. They're a bit looser this time; no custom enums, ignoring things that would need a refactor to fix.