cleanup: fix indentation
add & cleanup docstrings