Update prime_number.py