Clean up docstrings and a comment.