cleanup
cleanup
correct docstring
example tests for two tasks