Add docstrings and TODO.