Update prime.py