Updated for HW4 section 3-9