doctest'ing prime number example