diff --git a/docs/codes/index.md b/docs/codes/index.md
index 8f9593e..2e72c14 100644
--- a/docs/codes/index.md
+++ b/docs/codes/index.md
@@ -52,6 +52,17 @@ A differentiable physics-informed ecosystem model
 ![Differentiable Parameter Learning](../assets/project-figures/Tsai_etal2021.png){align="left" width="120"}
 Differentiable Parameter Learning
 
+  </div>
+</div>
+<div class="result" markdown>
+  <div class="grid cards" markdown>
+
+-   [$\delta$ differentiable stream temperature (LSTM+SNTEMP) __(Rahmani et al. 2023)__][Rahmani_et_al_2023.md]
+   
+    ---
+![Scatterplots comparing baseflow ratio predicted by four models with different structural priors](../assets/project-figures/Fig5_Rahmani2023.png)
+{align="left" width="120"}
+CONUS scale pearson correlation between baseflow estimation of our model and GAGES-II estimates of baseflow.
 
   </div>
 </div>
@@ -108,20 +119,8 @@ Time series of the forward model and the models integrating snow water equivalen
 -   [LSTM stream temperature model __(Rahmani et al. 2021)__][Rahmani_et_al_2021.md]
    
     ---
-![Performance of models with observed streamflow, simulated streamflow, no streamflow data, and a locally fitted auto-regressive model](../assets/project-figures/Rahmani_et_al_2021.PNG){align="left" width="140"}
-CONUS scale aggregated metrics of stream temperature models for the test period. 
-
-  </div>
-</div>
-<div class="result" markdown>
-  <div class="grid cards" markdown>
-
--   [$\delta$ differentiable stream temperature (LSTM+SNTEMP) __(Rahmani et al. 2023)__][Rahmani_et_al_2023.md]
-   
-    ---
-![Scatterplots comparing baseflow ratio predicted by four models with different structural priors](../assets/project-figures/Fig5_Rahmani2023.jpg)
-{align="left" width="150"}
-CONUS scale pearson correlation between baseflow estimation of our model and GAGES-II estimates of baseflow.
+![Performance of models with observed streamflow, simulated streamflow, no streamflow data, and a locally fitted auto-regressive model](../assets/project-figures/Rahmani_et_al_2021.png){align="left" width="120"}
+CONUS scale aggregated metrics of stream temperature models for the test period.
 
   </div>
 </div>