Update prime.cpp