diff --git a/sw/plot_heatmaps.py b/sw/plot_heatmaps.py
index c596b35..08c0959 100644
--- a/sw/plot_heatmaps.py
+++ b/sw/plot_heatmaps.py
@@ -18,38 +18,38 @@ def format_yticks(previous_yticks):
 def main():
     sns.set_theme()
 
-    titles = [
-        "$n=7$, $m=4$",
-        "$n=31$, $m=20$",
-        "$n=31$, $m=5$",
-        "$n=96$, $m=48$",
-        "$n=204$, $m=102$",
-        "$n=408$, $m=204$",
-    ]
-
-    filenames = [
-        "sim_results/2d_dec_fails_w_log_k_lin_bch_7_4alist.csv",
-        "sim_results/2d_dec_fails_w_log_k_lin_bch_31_11alist.csv",
-        "sim_results/2d_dec_fails_w_log_k_lin_bch_31_26alist.csv",
-        "sim_results/2d_dec_fails_w_log_k_lin_963965alist.csv",
-        "sim_results/2d_dec_fails_w_log_k_lin_2043486alist.csv",
-        "sim_results/2d_dec_fails_w_log_k_lin_40833844alist.csv",
-    ]
-
     # titles = [
     #     "$n=7$, $m=4$",
     #     "$n=31$, $m=20$",
     #     "$n=31$, $m=5$",
-    #     "$n=96$, $m=48$"
+    #     "$n=96$, $m=48$",
+    #     "$n=204$, $m=102$",
+    #     "$n=408$, $m=204$",
     # ]
     #
     # filenames = [
-    #     "sim_results/2d_dec_fails_w_log_k_lin_zoomed_in_bch_7_4alist.csv",
-    #     "sim_results/2d_dec_fails_w_log_k_lin_zoomed_in_bch_31_11alist.csv",
-    #     "sim_results/2d_dec_fails_w_log_k_lin_zoomed_in_bch_31_26alist.csv",
-    #     "sim_results/2d_dec_fails_w_log_k_lin_zoomed_in_963965alist.csv",
+    #     "sim_results/2d_dec_fails_w_log_k_lin_bch_7_4alist.csv",
+    #     "sim_results/2d_dec_fails_w_log_k_lin_bch_31_11alist.csv",
+    #     "sim_results/2d_dec_fails_w_log_k_lin_bch_31_26alist.csv",
+    #     "sim_results/2d_dec_fails_w_log_k_lin_963965alist.csv",
+    #     "sim_results/2d_dec_fails_w_log_k_lin_2043486alist.csv",
+    #     "sim_results/2d_dec_fails_w_log_k_lin_40833844alist.csv",
     # ]
 
+    titles = [
+        "$n=7$, $m=4$",
+        "$n=31$, $m=20$",
+        "$n=31$, $m=5$",
+        "$n=96$, $m=48$"
+    ]
+
+    filenames = [
+        "sim_results/2d_dec_fails_w_log_k_lin_zoomed_in_bch_7_4alist.csv",
+        "sim_results/2d_dec_fails_w_log_k_lin_zoomed_in_bch_31_11alist.csv",
+        "sim_results/2d_dec_fails_w_log_k_lin_zoomed_in_bch_31_26alist.csv",
+        "sim_results/2d_dec_fails_w_log_k_lin_zoomed_in_963965alist.csv",
+    ]
+
     fig, axes = plt.subplots(2, 3, squeeze=False)
 
     fig.suptitle("SNR = 3dB")