Create prime number.py