update CBB520HW2_part2.java class