Finish prime.py and link it in the README