Updates to fix week 2 hw