Turn this into a doctest block