Create compute_prime.py