Small changes
Problem in the Confusion Matrix = FIXED