better docstr - thanks @cyli
better test names and docstring - thanks @cyli