Add prime_generator