Add type annotations (for mypy)

[mverleg]

This will probably have to wait until python 2 support is discontinued (probably in 2020).