diff --git a/sw/simulate_2d_dec_fails.py b/sw/simulate_2d_dec_fails.py
index e237a8a..c38d501 100644
--- a/sw/simulate_2d_dec_fails.py
+++ b/sw/simulate_2d_dec_fails.py
@@ -8,7 +8,8 @@ from timeit import default_timer
 from utility import codes, noise, misc
 from utility.simulation.simulators import GenericMultithreadedSimulator
 
-from cpp_modules.cpp_decoders import ProximalDecoder
+# from cpp_modules.cpp_decoders import ProximalDecoder
+from cpp_modules.cpp_decoders import ProximalDecoder_31_5 as ProximalDecoder
 
 
 def task_func(params):
@@ -17,6 +18,7 @@ def task_func(params):
     decoder, num_iterations, x_bpsk, SNR, n, k = params
 
     dec_fails = 0
+
     for i in range(num_iterations):
         x = noise.add_awgn(x_bpsk, SNR, n, k)
         x_hat, num_iter = decoder.decode(x)
@@ -72,10 +74,13 @@ def main():
 
     sim_name = "w_log_k_lin_zoomed_in"
 
-    H_file = "96.3.965.alist"
+    # H_file = "96.3.965.alist"
+    # H_file = "204.3.486.alist"
+    # H_file = "408.33.844.alist"
+    H_file = "BCH_31_26.alist"
     SNR = 3
 
-    num_iterations = 1000
+    num_iterations = 10000
     omegas = np.logspace(-0.3, -2.82, 40)
     Ks = np.ceil(np.linspace(10 ** 1.3, 10 ** 2.3, 40)).astype('int32')