diff --git a/fig/fig0_overview2.pdf b/fig/fig0_overview2.pdf
index c6b5333..9a33788 100755
Binary files a/fig/fig0_overview2.pdf and b/fig/fig0_overview2.pdf differ
diff --git a/fig/fig0_overview2_tikz.tex b/fig/fig0_overview2_tikz.tex
index d5bb6b4..25172f2 100755
--- a/fig/fig0_overview2_tikz.tex
+++ b/fig/fig0_overview2_tikz.tex
@@ -9,8 +9,8 @@
   chan/.style={draw,diamond,aspect=1.4,fill=gray!25,inner sep=0.8pt},
   rxb/.style={box},
   mult/.style={draw,circle,inner sep=0.4pt,minimum size=4mm},
-  sum/.style={draw,rounded corners=2.5mm,fill=cyan!20,minimum width=12mm,minimum height=26mm},
-  dec/.style={draw,rounded corners=2pt,minimum width=47mm,minimum height=6mm,inner sep=2pt},
+  sum/.style={draw,rounded corners=2.5mm,fill=cyan!20,minimum width=12mm,minimum height=36mm},
+  dec/.style={draw,rounded corners=2pt,minimum width=53mm,minimum height=6mm,inner sep=2pt},
   lbl/.style={font=\small\bfseries,align=center},
   arr/.style={-stealth,semithick}]
 
@@ -42,7 +42,7 @@
 \foreach \c in {x,f,o,h,y}{\node at ($(\c 2)!0.5!(\c U)$) {$\vdots$};}
 
 % superposition
-\node[sum,right=9mm of y2.east,yshift=-3.5mm] (sig)
+\node[sum,right=9mm of y2.east,yshift=-2mm] (sig)
   {$\displaystyle\sum_{v=1}^{U}$};
 \draw[arr] (y1.east) -- (y1.east -| sig.west);
 \draw[arr] (y2.east) -- (y2.east -| sig.west);
@@ -54,7 +54,7 @@
 \node[lbl] at ([yshift=-4.5mm]sig.south) {Received Signal\\(Superposition $\tilde{\mathbf{y}}$)};
 
 % ---------------- right panel: UWCA decoder ----------------
-\node[dec,fill=violet!15,right=15mm of sig.east,yshift=19mm] (mask)
+\node[dec,fill=violet!15,right=26mm of sig.east,yshift=19mm] (mask)
   {Soft Masks $\{\tilde{\mathbf{m}}_i\}_{i=1}^{U}$};
 \node[dec,fill=blue!12,below=2.2mm of mask,align=center] (proj)
   {K/V Projection $(\mathbf{W}_K,\mathbf{W}_V)$\\[-1pt] Queries $\{\mathbf{q}_u\}$};
@@ -70,21 +70,21 @@
 \draw[arr] (attn) -- (plus);
 \draw[arr] (plus) -- (norm);
 \draw[arr] (norm) -- (out);
-\draw[arr] (sig.east) -- node[above,font=\footnotesize]{$\tilde{\mathbf{y}}$} ++(6mm,0) |- (mask.west);
+\draw[arr] (sig.east) -- node[above,pos=0.45,font=\small]{$\tilde{\mathbf{y}}$} ++(13mm,0) |- (mask.west);
 \draw[arr,dashed,blue] (mask.east) -- ++(4mm,0) |- (plus.east)
-  node[pos=0.25,right,font=\footnotesize,align=left]{skip:\\$\tilde{\mathbf{y}}\odot\tilde{\mathbf{m}}_u$};
+  node[pos=0.78,above,font=\footnotesize]{skip: $\tilde{\mathbf{y}}\odot\tilde{\mathbf{m}}_u$};
 
 % decoder panel frame
 \begin{scope}[on background layer]
-\node[draw=blue!60,dashed,rounded corners=2mm,fill=blue!5,fit=(mask)(out)(attn),
-      inner xsep=10mm,inner ysep=2.5mm] (panel) {};
+\node[rounded corners=2mm,fill=gray!12,fit=(mask)(out)(attn),
+      inner xsep=15mm,inner ysep=4.5mm] (panel) {};
 \end{scope}
-\node[anchor=south,font=\small\bfseries\color{blue!60!black},inner sep=1.5pt] at (panel.north)
+\node[anchor=south,font=\small\bfseries\color{black!75},inner sep=2pt] at (panel.north)
   {UWCA Decoder (per-user, shared weights $\theta$)};
 
 % MAML brace
 \draw[decorate,decoration={brace,mirror,amplitude=2mm},blue!60!black,thick]
-  ([yshift=-1.2mm]panel.south west) -- ([yshift=-1.2mm]panel.south east)
+  ([yshift=-1.5mm]panel.south west) -- ([yshift=-1.5mm]panel.south east)
   node[midway,below=2mm,font=\small\color{blue!60!black}]
   {MAML meta-training over SNR tasks $\{\mathcal{T}_k\}_{k=1}^{K}$};
 \end{tikzpicture}