docstring fix + test clean up for #9