fix docstrings
add TODO list
update docstrings