Add some simple docstrings and a TODO comment.