added return to prim visualization