wrote predict_forest and skeleton of test function