Would be good to have consistent type annotations: typing in docstrings should be reflected in the annotations, and be brought up to date with latest pep, eg Dict->dict