Fixed probability output, added docstrings, and updated classification p...

...lotting example

I need to merge this to finish up the rest of the stuff for my talk, so I will do that now.

cc @eickenberg for completeness.