diff --git a/R/idiomclassification_mk_pf.R b/R/idiomclassification_mk_pf.R
index 99025b5..82516a3 100644
--- a/R/idiomclassification_mk_pf.R
+++ b/R/idiomclassification_mk_pf.R
@@ -61,13 +61,12 @@
 # optional resampling with smote
 
 smoted.data <- SMOTE(fmla, subset(train, select = c("CO_IDIOM", vars)), perc.over = 1200, perc.under = 100)
-rf_classifier = randomForest(fmla, smoted.data, ntree=100, mtry=4, importance=TRUE)
-prediction_for_table <- predict(rf_classifier,test %>% select(-CO_IDIOM))
+rf_classifier = randomForest(fmla, smoted.data, ntree=200, importance=TRUE)
+prediction_for_table <- predict(rf_classifier,test %>% select(-CO_IDIOM), cutoff=c(0.8,0.2))
 confusion <- table(predicted=prediction_for_table, observed=test$CO_IDIOM)
 conf <- confusionMatrix(confusion, positive = "1")
 print(conf)
 # Using estimates by random forest on entire dataset
-
 library(randomForest)
 rf_classifier_full = randomForest(fmla, data=ngramme, ntree=100, mtry=2, importance=TRUE, cutoff=c(0.8,0.2))
 rf_classifier_full