update and cleanup
update
Docstrings.