Add docstrings and todos.