prime.py updated