Docstrings updated with type info