Cleaned up doctest