Added classification algorithm to solver