make_prediction now uses the columns to drop as argument to make sure it is the same as the test functions