Updated docstring, better test.