Bit of a dirty hack, but I managed to fix some stupid behaviour of the trees for labels 1 and 5. Also, DecisionTree now takes a label parameter. I also started writing the report on Writelatex, outside this git repo