updated TEST.py, PRIME1.py