Updated my prim.py

… forgot that I was adding the edges such that the same might appear
twice. Should be solved now.