TODO: Prim’s algorithm