Update Prim.py