iteration in python programming - Search