modify TextClassifier. Not finished yet.