generator didn't work, replacing with comprehension.