[WIP] Modified the confusion matrix example

Ah. I see your note below.