Update prime_num.py