Bug fix to data generation script
Merge pull request #75 from ramonsnir/master

Bug fix to data generation script