Adding import itertools