Fixed itertools import