More changes to handle the MCC being used in the variable improtance when there are only two classes