Update get_prime.py