Update is_prime.py