Partially completed the TrainTests junit testing class