Added code for itertools