Update primes.py