Drop itertools.count(1)

A generator here is overkill. A simple int would work just as well.