add prim.py