Modify TrainProfile.java, it is still under developed