Making code challenge 2 clearer