Added random forest model, test set