fix an logic flaws in the last question in challenging set"