Add types to docstrings