diff --git a/docs/src/assets/lbdn-mnist/lbdn_mnist.svg b/docs/src/assets/lbdn-mnist/lbdn_mnist.svg
index 8383238..71f4c61 100644
--- a/docs/src/assets/lbdn-mnist/lbdn_mnist.svg
+++ b/docs/src/assets/lbdn-mnist/lbdn_mnist.svg
@@ -2,531 +2,411 @@
diff --git a/docs/src/assets/lbdn-mnist/lbdn_mnist_robust.svg b/docs/src/assets/lbdn-mnist/lbdn_mnist_robust.svg
index f93f9d1..5d4d568 100644
--- a/docs/src/assets/lbdn-mnist/lbdn_mnist_robust.svg
+++ b/docs/src/assets/lbdn-mnist/lbdn_mnist_robust.svg
@@ -2,213 +2,219 @@