Removed unused itertools import