Removed itertools unused import