Update TrainingPartTwo.java

updated with comment