Create primes.py