Create prime.py