Add comments to challenge 4 solution