some progress on week 2 homework