HW2 part2 partially done