Forgot to include chain from itertools while using it