Added update to classification model to make correct plots during training
Merge branch 'master' of https://github.com/rjagerman/aidu