removed unused classes for predicted class