adding test for  in week 2