add prime.py