Update the test function in BaggingModel.java