add itertools import