minor fix on data generator
Fixed the issues with data generator
Merge branch 'master' of https://github.com/boyshawn/CS4221