Create prime.py

Learning python for the first time