TYP: Add return types to some top-level func
Adds return type hints to some top-level funcs. I will take the rest over 1-2 more PRs.

Having return types typed up will be good for the user experience (better piping etc.).