added challenge 1 code