Code added for the classifier