MAINT/DOC: Add type information to arguments

[zoj613]

Add some typing information to make arguments a little easier to read (maybe). I tend to prefer stub files (e.g. see here) over in-place typing since there is less distracting lines of code. However it means the file count doubles because every python source is accompanied by a .pyi file.