diff --git a/artifacts/results/images/mnist_eccco.png b/artifacts/results/images/mnist_eccco.png
index acc03a90aac1bb5cd322167b79a0c53df280a939..79810ccc202670e45974d390d3542c5bab0c52ec 100644
Binary files a/artifacts/results/images/mnist_eccco.png and b/artifacts/results/images/mnist_eccco.png differ
diff --git a/artifacts/results/images/mnist_generated_JEM Ensemble.png b/artifacts/results/images/mnist_generated_JEM Ensemble.png
index c1ab0d749b8ea897bef690a675f588a484cad775..2ae769456baf5b76a0a5175b3af49e0fab8c2516 100644
Binary files a/artifacts/results/images/mnist_generated_JEM Ensemble.png and b/artifacts/results/images/mnist_generated_JEM Ensemble.png differ
diff --git a/artifacts/results/images/mnist_generated_JEM.png b/artifacts/results/images/mnist_generated_JEM.png
index 55986532a9d3fb152e7a0d30f06c4c16465cfdfb..6af99ea744e854095b29c15716c4476b9507b18c 100644
Binary files a/artifacts/results/images/mnist_generated_JEM.png and b/artifacts/results/images/mnist_generated_JEM.png differ
diff --git a/artifacts/results/images/mnist_generated_MLP Ensemble.png b/artifacts/results/images/mnist_generated_MLP Ensemble.png
index 4b61caf6a16da195d882ebb4f58b7822497bd08b..520a0b572e39675328cccf3e174cbaddda51198d 100644
Binary files a/artifacts/results/images/mnist_generated_MLP Ensemble.png and b/artifacts/results/images/mnist_generated_MLP Ensemble.png differ
diff --git a/artifacts/results/images/mnist_generated_MLP.png b/artifacts/results/images/mnist_generated_MLP.png
index 3a12c44c406fb2495e6f9534e4f77cd659e9ab08..34f1da9ada25113e904a6fde41853dadd84e8ba0 100644
Binary files a/artifacts/results/images/mnist_generated_MLP.png and b/artifacts/results/images/mnist_generated_MLP.png differ
diff --git a/artifacts/results/mnist_architectures.jls b/artifacts/results/mnist_architectures.jls
index ff591e713aa432da892c1adead65f15de1c253b7..4054e47c244cf42615793282663b50e07abf4755 100644
Binary files a/artifacts/results/mnist_architectures.jls and b/artifacts/results/mnist_architectures.jls differ
diff --git a/artifacts/results/mnist_model_performance.csv b/artifacts/results/mnist_model_performance.csv
index c61e7d1849d84f1250982e1555348ad85433b9f5..574a21ce7d69e0b205f395e8be9186d24f1f05ee 100644
--- a/artifacts/results/mnist_model_performance.csv
+++ b/artifacts/results/mnist_model_performance.csv
@@ -1,5 +1,5 @@
 acc,precision,f1score,mod_name,dataname
-0.9154,0.9156058593092286,0.9144154048006502,JEM Ensemble,MNIST
-0.9651,0.9649131785895403,0.9647206151544168,MLP,MNIST
-0.9745,0.9743410428044881,0.9743191770867725,MLP Ensemble,MNIST
-0.8533999999999999,0.8673648922304751,0.8529767185660582,JEM,MNIST
+0.8985,0.8986458930056993,0.8970545599029716,JEM Ensemble,MNIST
+0.9402,0.9404816585665723,0.9395994247759306,MLP,MNIST
+0.9423,0.9418414616466535,0.9417117851276802,MLP Ensemble,MNIST
+0.8054,0.8287830565193942,0.8039356667514831,JEM,MNIST
diff --git a/artifacts/results/mnist_model_performance.jls b/artifacts/results/mnist_model_performance.jls
index f5b4a01a9f26fb26505c1bd1e01d82743b7072a5..6d60c57129ae9269ccb884811e21d2240cb9100a 100644
Binary files a/artifacts/results/mnist_model_performance.jls and b/artifacts/results/mnist_model_performance.jls differ
diff --git a/artifacts/results/mnist_models.jls b/artifacts/results/mnist_models.jls
index 4e21686abb0795da0f7f11bc292baf289706df3b..8c4687d10216f753591d636241c5eef485878d00 100644
Binary files a/artifacts/results/mnist_models.jls and b/artifacts/results/mnist_models.jls differ
diff --git a/artifacts/results/mnist_vae.jls b/artifacts/results/mnist_vae.jls
index a1705aec7bdda962a4147b6aae68f8c2a3991828..1a74f76e4958d83d3fc08c1fe1c247292e07f297 100644
Binary files a/artifacts/results/mnist_vae.jls and b/artifacts/results/mnist_vae.jls differ
diff --git a/artifacts/results/mnist_vae_weak.jls b/artifacts/results/mnist_vae_weak.jls
index c0d6c2a6eb82fa5d9dfc10d86de3aaca234bf441..2100c22bdb104119ce7eb024a9d24da877da7105 100644
Binary files a/artifacts/results/mnist_vae_weak.jls and b/artifacts/results/mnist_vae_weak.jls differ
diff --git a/notebooks/mnist.qmd b/notebooks/mnist.qmd
index bc4da42932630634e0381dec321c57f7e098abe5..4a8448348c1111347657051e51db4740207a329f 100644
--- a/notebooks/mnist.qmd
+++ b/notebooks/mnist.qmd
@@ -168,7 +168,7 @@ _retrain = true
 _regen = true
 
 # Data:
-n_obs = nothing
+n_obs = 10000
 counterfactual_data = load_mnist(n_obs)
 counterfactual_data.X = pre_process.(counterfactual_data.X)
 counterfactual_data.generative_model = vae
@@ -207,7 +207,7 @@ sampler = ConditionalSampler(
     input_size=(input_dim,), 
     batch_size=10,
 )
-α = [1.0,1.0,25e-3]      # penalty strengths
+α = [1.0,1.0,1e-2]      # penalty strengths
 ```
 
 ```{julia}