minor change to challenge 6