Update HW2

Still need edit