Update prim.cpp