Include prime.py and update README.md