clean repo, fix docstring