Updated it to use Prim's

Hopefully it works?