Changes in week3/2 challenge