revise lib/prime.py