Update docstrings, comments