Improve docs and add a doctest.