Clean Up

Removed unused methods/classes etc.
Main problem with confusion matrix was that it tested against training
data. Whoops!