[DM HW3] Prob 6 first draft