import itertools instead of from itertools import *