Update prime number.py