Clean up doctests