src code after adding the classifier