Poprawienie błędu w mapowaniu klas i dodanie confusion matrix