Fix itertools import - something else going bad on Python3?