added predictNewClasses function