q118, modify test_prime