Add docstrings and type hints

[joosthooz]

Many functions are currently documented with an ordinary comment line above them; these should be converted to proper docstrings. While we're at it, we should also add type hints to the functions.