Before revision:  In this paper we extend that work, by (1) improving the quality of the training data and (2) applying BERT-based models to the problem.
After revision:  In this paper we extend that work, by (1) improving the quality of the training data and train three BERT-based models to the problem.