fixed error in week 3