I added a histogram display, and fixed an issue when re-running the
training.
On top of that, I altered the KL distance method to look in all of the
tree instead of just the subtrees.