Create is_prime.py