Fixed prime.py, just cleaned it up