cleaning up
docstring changes
working tests