Prims algo code