Ah, I misunderstood. I thought you were suggesting that the for loop would also be in `__init__`. I'll push an update shortly.