almost done with HW2