Add docstrings, other comments