diff --git a/figures/fully-connected-IID-vs-non-IID.png b/figures/fully-connected-IID-vs-non-IID.png
index ee3483dc5763b17e85fd96864cd7d5f1b388d8fd..77e8914712046e8784c1fa23cf018c545a830713 100644
Binary files a/figures/fully-connected-IID-vs-non-IID.png and b/figures/fully-connected-IID-vs-non-IID.png differ
diff --git a/figures/grid-IID-vs-non-IID.png b/figures/grid-IID-vs-non-IID.png
index 5dfbfcba32050f0f19730b04d7587ed7736487b6..b809feab14c57cc1da97954c8e9088ffdfadb155 100644
Binary files a/figures/grid-IID-vs-non-IID.png and b/figures/grid-IID-vs-non-IID.png differ
diff --git a/figures/ring-IID-vs-non-IID.png b/figures/ring-IID-vs-non-IID.png
index fe50e659de791a26874e7eac1ddd84e49a848768..6faa52fd396b8005ad428867d135175d891779ac 100644
Binary files a/figures/ring-IID-vs-non-IID.png and b/figures/ring-IID-vs-non-IID.png differ
diff --git a/main.tex b/main.tex
index 4a3663cf3f7be05fc9894ed887ad7731fc2184a6..f4976ff26268d15a18e78e7671e245f8f22f8c1c 100644
--- a/main.tex
+++ b/main.tex
@@ -141,22 +141,29 @@ The rest of this paper is organized as follows. We first present the problem, as
 
 \begin{figure}
      \centering
-     \begin{subfigure}[b]{0.34\textwidth}
+     
+     % From directory results/mnist
+     % python ../../../learn-topology/tools/plot_convergence.py ring/iid/all/2021-03-30-16:07:06-CEST ring/non-iid/all/2021-03-30-16:07:03-CEST --add-min-max --legend 'lower right' --yaxis test-accuracy --labels '100 nodes IID' '100 nodes non-IID' --save-figure ../../figures/ring-IID-vs-non-IID.png --font-size 20
+     \begin{subfigure}[b]{0.31\textwidth}
          \centering
          \includegraphics[width=\textwidth]{figures/ring-IID-vs-non-IID}
 \caption{\label{fig:ring-IID-vs-non-IID} Ring: (almost) minimal connectivity.}
      \end{subfigure}
-     \hfill
-     \begin{subfigure}[b]{0.27\textwidth}
+     \quad
+    % From directory results/mnist
+     % python ../../../learn-topology/tools/plot_convergence.py grid/iid/all/2021-03-30-16:07:01-CEST grid/non-iid/all/2021-03-30-16:06:59-CEST --add-min-max --legend 'lower right' --yaxis test-accuracy --labels '100 nodes IID' '100 nodes non-IID' --save-figure ../../figures/grid-IID-vs-non-IID.png --font-size 20
+     \begin{subfigure}[b]{0.31\textwidth}
          \centering
          \includegraphics[width=\textwidth]{figures/grid-IID-vs-non-IID}
 \caption{\label{fig:grid-IID-vs-non-IID} Grid: intermediate connectivity.}
      \end{subfigure}
-     \hfill
-     \begin{subfigure}[b]{0.34\textwidth}
+     \quad
+         % From directory results/mnist
+     % python ../../../learn-topology/tools/plot_convergence.py fully-connected/iid/all/2021-03-30-16:07:20-CEST fully-connected/all/2021-03-10-09:25:19-CET  --add-min-max --legend 'lower right' --yaxis test-accuracy --labels '100 nodes IID' '100 nodes non-IID' --save-figure ../../figures/fully-connected-IID-vs-non-IID.png --font-size 20
+     \begin{subfigure}[b]{0.31\textwidth}
          \centering
          \includegraphics[width=\textwidth]{figures/fully-connected-IID-vs-non-IID}
-\caption{\label{fig:fully-connected-IID-vs-non-IID} Fully-connected: maximal connectivity.}
+\caption{\label{fig:fully-connected-IID-vs-non-IID} Fully-connected: max connectivity.}
      \end{subfigure}
         \caption{IID vs non-IID Convergence Speed. Thin lines are the minimum
         and maximum accuracy of individual nodes. Bold lines are the average