finished part-1 of HW4