Forgot to import itertools