FlattenIterator: Refactor