HW4 part6 almost there