Update prime_num2.py