look up how to keep track of iteration count in list comprehension