Update mst.py