Remove unused method
Add docstrings
Label confusion matrix