updated wk4 challenge 2