Fixed errors in RunTrainPredict