start fixing up docs and doctest