HW Lesson 4 Problem Set