Tweak docstring and extra comparison tests