finished part-2 of HW4