Added code for predicting with randomForest