adding some changes to challenge 3