Use generator for list of treebank files

Should increase speed.
Merge pull request #1 from smargh/patch-1

Use generator for list of treebank files