Fixed up the doctest so it's a little cleaner looking.