Adding weight capabilities for MST