updated HW4
updated HW4