Create prime.py

basic prime number checker