week 6 problem set 2 almost done