Finished RunTrainPredict and fixed errors