Don't think you need the itertools import here.