HW5 - up to question 7