fixed doctest issues, I think