Added simple version of prims, edited normal one