Add prime.py