HW3 Update up til Problem 1