Update mriTrain.m