diff --git a/src/classes/Heuristic.py b/src/classes/Heuristic.py
index 6d71790..86d449c 100644
--- a/src/classes/Heuristic.py
+++ b/src/classes/Heuristic.py
@@ -77,18 +77,17 @@ class WeightHeuristicEngine(HeuristicEngine):
 
     def get_weight(self, board, player):
         score = 0
-        size = self.options['size']
-        w = [[0 for _ in range(size)] for _ in range(size)]
         for pos_x in range(self.options['size']):
             for pos_y in range(self.options['size']):
+                # Get player at x,y
                 p = board._board[pos_x][pos_y]
+
                 if p == player:
                     score += self.weights[pos_x][pos_y]
-                    w[pos_x][pos_y] = self.weights[pos_x][pos_y]
 
-                elif p != player and p != board._EMPTY:
+                elif p != board._EMPTY:
                     score -= self.weights[pos_x][pos_y]
-                    w[pos_x][pos_y] = -self.weights[pos_x][pos_y]
+
         return score
 
     """