Add filter based on randomForestSRC