Using python generators with lots of data