fix docstring
update TODO.txt