Added stuff for number 6 on HW 2