Use generator for list of treebank files

Should increase speed.