Modifying iterables with Itertools - Python