Update docs after doctest fixes