Update prime_generation.cpp