HW#4 Part 2 done