typo fixes + added a TODO
docstring updates