Fixed bug datagenerator
Merge branch 'master' of https://github.com/SteveLovesPudding/ML_FinalProject.git