HW5 part2 part4 done!