Completed part 2 of HW3