Now handling cases where the test data contains new labels more gracefully. Shows a WARNING instead of an ERROR