Edit: primes.py