update TODO - docstrings