Make doctest nicer