Add and amend docstrings mostly