* add strict type annotations * apply suggestions Co-authored-by: Erik Montnemery <erik@montnemery.com>