Create TrainProblem.java