Adding another gen_primes function