replaced my generator with itertools count()