Update the prim implementation