Minor modification to set1, challenge 1.