Add some improved docstrings