Made a classification factory. And integrated it with the rest of the application.