HW02: part 6 finished