Quick update to challenge 4