Update mst.cpp