diff --git a/TEST_2.py b/TEST_2.py
index 7b97f7c7af115dae3c16e129e4da90ed918779cb..cbe7d1638c1ff423688c4408a726971eef1fadf6 100644
--- a/TEST_2.py
+++ b/TEST_2.py
@@ -19,12 +19,12 @@ STR = Structure2DPlaneStress(mesh_elements, mesh_points,
                               poisson_ratio = 0.3,
                               thickness = 0.1)
 
-NODES = STR.nodes
 
+STRUCT = list()
+STRUCT.append(STR)
+NODES = STR.nodes
 ELEMENTS = STR.elements
 
-STR.nodes[2].force_2d = Force2D([1,1])
 
+STR.nodes[2].force_2d = Force2D([1,1])
 STR.solve()
-A = list()
-A.append(STR)
\ No newline at end of file
diff --git a/fem_2d.py b/fem_2d.py
index 814674c6323002cd294f5b599248ff1c5b3015b3..40981dbc12aad1e7ba1ea248f4485971bbbd2f0f 100644
--- a/fem_2d.py
+++ b/fem_2d.py
@@ -176,11 +176,14 @@ class Structure2DPlaneStress:
             node_ids = self.elements[element_id].node_ids
             k_e = self.elements[element_id].get_k_mat(self.thickness,
                                                       self.c_mat)
-            # First [0,0] component of k_e
-            
+            for (i, j) in ((0, 0), (0, 1), (1, 0), (1, 1)):
+                k_global[node_ids[i], node_ids[j]] = (k_global[node_ids[i],
+                                                               node_ids[j]] +
+                                                      k_e[i, j])
+
+
+
 
-            
-    
 
 
 '''