diff --git a/augment.py b/augment.py
index b57cb78..1931ae2 100644
--- a/augment.py
+++ b/augment.py
@@ -14,7 +14,7 @@
 os.makedirs(images_folder, exist_ok=True)
 
 # Set the ROI size and step
-roi_size = (256, 256)
+roi_size = (512, 512)
 roi_step = 128
 
 # Get the list of file names in the mask_org folder
diff --git a/figure01.md b/figure01.md
index d02348c..9f14c71 100644
--- a/figure01.md
+++ b/figure01.md
@@ -217,10 +217,34 @@ tensorboard --logdir=.
 network](./repo_img/tensorboard.png)
 
 Click the **images** tab of the Tensorboard to inspect the visual output
-of the training. This is in the beginning: ![inspect the
-output](./repo_img/tensorboard_img.png)
+of the training.
 
-Here:
+<div id="fig-training">
+
+<table style="width:50%;">
+<colgroup>
+<col style="width: 50%" />
+</colgroup>
+<tbody>
+<tr class="odd">
+<td style="text-align: center;"><div width="50.0%"
+data-layout-align="center">
+<p><img src="./repo_img/tensorboard_img.png" id="fig-begin"
+style="width:40.0%" alt="begin" /> <img
+src="./repo_img/tensorboard_later.png" id="fig-later"
+style="width:40.0%" alt="later" /></p>
+</div></td>
+</tr>
+</tbody>
+</table>
+
+Figure 1: Training monitoring. Left is in the beginning right is later
+during training. Top row is input, middle is output of network and
+bottom is the reference (what the network should aim for).
+
+</div>
+
+In detail:
 
 - `net_input` is the input images. Notice a cell is only really present
   in the first out of three.
@@ -261,6 +285,6 @@ alt="(b) output" /></p>
 </tbody>
 </table>
 
-Figure 1: Segmentation results.
+Figure 2: Segmentation results.
 
 </div>
diff --git a/figure01.qmd b/figure01.qmd
index e17cd68..df9ad94 100644
--- a/figure01.qmd
+++ b/figure01.qmd
@@ -174,10 +174,16 @@ tensorboard --logdir=.
 
 
 Click the **images** tab of the Tensorboard to inspect the visual output of the training.
-This is in the beginning:
-![inspect the output](./repo_img/tensorboard_img.png)
 
-Here:
+
+::: {#fig-training layout-ncol=2}
+![begin](./repo_img/tensorboard_img.png){#fig-begin width=40%}
+![later](./repo_img/tensorboard_later.png){#fig-later width=40%} 
+
+Training monitoring. Left is in the beginning right is later during training. Top row is input, middle is output of network and bottom is the reference (what the network should aim for).
+:::
+
+In detail:
 
 - `net_input` is the input images. Notice a cell is only really present in the first out of three.
 - `net_output0` is the current output from the network.
diff --git a/models/dapi/logs/images/events.out.tfevents.1695132333.UU-N9KR2WKWCL.637.0.v2 b/models/dapi/logs/images/events.out.tfevents.1695143237.UU-N9KR2WKWCL.18242.0.v2
similarity index 69%
rename from models/dapi/logs/images/events.out.tfevents.1695132333.UU-N9KR2WKWCL.637.0.v2
rename to models/dapi/logs/images/events.out.tfevents.1695143237.UU-N9KR2WKWCL.18242.0.v2
index 12b091d..decf742 100644
Binary files a/models/dapi/logs/images/events.out.tfevents.1695132333.UU-N9KR2WKWCL.637.0.v2 and b/models/dapi/logs/images/events.out.tfevents.1695143237.UU-N9KR2WKWCL.18242.0.v2 differ
diff --git a/models/dapi/logs/train/events.out.tfevents.1695132333.UU-N9KR2WKWCL.637.1.v2 b/models/dapi/logs/train/events.out.tfevents.1695132333.UU-N9KR2WKWCL.637.1.v2
deleted file mode 100644
index 3411dd6..0000000
Binary files a/models/dapi/logs/train/events.out.tfevents.1695132333.UU-N9KR2WKWCL.637.1.v2 and /dev/null differ
diff --git a/models/dapi/logs/train/events.out.tfevents.1695143237.UU-N9KR2WKWCL.18242.1.v2 b/models/dapi/logs/train/events.out.tfevents.1695143237.UU-N9KR2WKWCL.18242.1.v2
new file mode 100644
index 0000000..bc93335
Binary files /dev/null and b/models/dapi/logs/train/events.out.tfevents.1695143237.UU-N9KR2WKWCL.18242.1.v2 differ
diff --git a/models/dapi/logs/validation/events.out.tfevents.1695132347.UU-N9KR2WKWCL.637.2.v2 b/models/dapi/logs/validation/events.out.tfevents.1695132347.UU-N9KR2WKWCL.637.2.v2
deleted file mode 100644
index 45359a2..0000000
Binary files a/models/dapi/logs/validation/events.out.tfevents.1695132347.UU-N9KR2WKWCL.637.2.v2 and /dev/null differ
diff --git a/models/dapi/logs/validation/events.out.tfevents.1695143251.UU-N9KR2WKWCL.18242.2.v2 b/models/dapi/logs/validation/events.out.tfevents.1695143251.UU-N9KR2WKWCL.18242.2.v2
new file mode 100644
index 0000000..45c2187
Binary files /dev/null and b/models/dapi/logs/validation/events.out.tfevents.1695143251.UU-N9KR2WKWCL.18242.2.v2 differ
diff --git a/models/dapi/thresholds.json b/models/dapi/thresholds.json
deleted file mode 100644
index 60382ea..0000000
--- a/models/dapi/thresholds.json
+++ /dev/null
@@ -1 +0,0 @@
-{"prob": 0.46905075238803473, "nms": 0.3}
\ No newline at end of file
diff --git a/models/dapi/weights_best.h5 b/models/dapi/weights_best.h5
index bf96e8a..7fbca8f 100644
Binary files a/models/dapi/weights_best.h5 and b/models/dapi/weights_best.h5 differ
diff --git a/models/dapi/weights_last.h5 b/models/dapi/weights_last.h5
deleted file mode 100644
index 4a590bd..0000000
Binary files a/models/dapi/weights_last.h5 and /dev/null differ
diff --git a/models/dapi/weights_now.h5 b/models/dapi/weights_now.h5
new file mode 100644
index 0000000..18fc366
Binary files /dev/null and b/models/dapi/weights_now.h5 differ
diff --git a/repo_img/tensorboard_img.png b/repo_img/tensorboard_img.png
index 5ced3b2..e5c925d 100644
Binary files a/repo_img/tensorboard_img.png and b/repo_img/tensorboard_img.png differ
diff --git a/repo_img/tensorboard_later.png b/repo_img/tensorboard_later.png
index 3efdcb4..67a6b20 100644
Binary files a/repo_img/tensorboard_later.png and b/repo_img/tensorboard_later.png differ