Added functionality to classify text