prime.py