diff --git a/notebooks/conformal.qmd b/notebooks/conformal.qmd
index 8a84a6cd9dd4867228672e5351b8743152adc3c4..5dcef400219e44811e5253e71aace478997e38ef 100644
--- a/notebooks/conformal.qmd
+++ b/notebooks/conformal.qmd
@@ -31,13 +31,8 @@ mach = machine(conf_model, X, y)
 fit!(mach)
 ```
 
-```{julia}
-contourf(mach.model, mach.fitresult, X, y; plot_set_size=true)
-```
-
 ## Counterfactual Explanation
 
-
 ```{julia}
 M = CCE.ConformalModel(conf_model, mach.fitresult)
 generator = CCE.ConformalGenerator()
@@ -45,8 +40,14 @@ generator = CCE.ConformalGenerator()
 
 ```{julia}
 x = select_factual(counterfactual_data,rand(1:size(counterfactual_data.X,2)))
-y = predict_label(M, counterfactual_data, x)[1]
-target = counterfactual_data.y_levels[counterfactual_data.y_levels .!= y][1]
+y_factual = predict_label(M, counterfactual_data, x)[1]
+target = counterfactual_data.y_levels[counterfactual_data.y_levels .!= y_factual][1]
+```
+
+```{julia}
+p1 = contourf(mach.model, mach.fitresult, X, y; plot_classification_loss=true, target=target, zoom=0)
+p2 = contourf(mach.model, mach.fitresult, X, y; plot_set_loss=true, zoom=0)
+plot(p1, p2, size=(800,320))
 ```
 
 ```{julia}