edited isprime.py example