Added prime.py script and tests for prime.py