Completed part 1 of HW4