Create prim.py