A few small fixes with newest prime.py library