Update TrainTestRun.java