completed challenge 2 in problem set 1