Small changes to the MVA training and some more plotting