diff --git a/sw/cpp/src/proximal.h b/sw/cpp/src/proximal.h
index 809c4da..65d6994 100644
--- a/sw/cpp/src/proximal.h
+++ b/sw/cpp/src/proximal.h
@@ -1,7 +1,5 @@
 #pragma once
 
-#define EIGEN_STACK_ALLOCATION_LIMIT 524288
-
 #include <Eigen/Dense>
 #include <bit>
 #include <iostream>
diff --git a/sw/cpp/src/python_interface.cpp b/sw/cpp/src/python_interface.cpp
index a9654af..df20027 100644
--- a/sw/cpp/src/python_interface.cpp
+++ b/sw/cpp/src/python_interface.cpp
@@ -1,3 +1,5 @@
+#define EIGEN_STACK_ALLOCATION_LIMIT 1048576
+
 #include "proximal.h"
 
 #include <pybind11/pybind11.h>
@@ -28,11 +30,17 @@ using namespace pybind11::literals;
 PYBIND11_MODULE(cpp_decoders, proximal) {
     proximal.doc() = "Proximal decoder";
 
-    DEF_PROXIMAL_DECODER("ProximalDecoder_7_4", 4, 7)
+    DEF_PROXIMAL_DECODER("ProximalDecoder_7_3", 3, 7)
+    DEF_PROXIMAL_DECODER("ProximalDecoder_31_20", 20, 31)
+    DEF_PROXIMAL_DECODER("ProximalDecoder_31_6", 6, 31)
     DEF_PROXIMAL_DECODER("ProximalDecoder_96_48", 48, 96)
     DEF_PROXIMAL_DECODER("ProximalDecoder_204_102", 102, 204)
+    DEF_PROXIMAL_DECODER("ProximalDecoder_408_204", 204, 408)
     DEF_PROXIMAL_DECODER("ProximalDecoder_Dynamic", Eigen::Dynamic,
                          Eigen::Dynamic)
 
-    py::register_exception<std::runtime_error>(proximal, "CppException");
+    py::register_exception<std::runtime_error>(
+        proximal, "CppException: std::runtime_error");
+    py::register_exception<std::bad_alloc>(proximal,
+                                           "CppException: std::bad_alloc");
 }
\ No newline at end of file