diff --git a/onn.py b/onn.py
index c05608a..5f0c05b 100755
--- a/onn.py
+++ b/onn.py
@@ -43,7 +43,7 @@ class SquaredHalved(ResidualLoss):
         return r
 
 class SomethingElse(ResidualLoss):
-    # generalizes Absolute and SquaredHalved
+    # generalizes Absolute and SquaredHalved.
     # plot: https://www.desmos.com/calculator/fagjg9vuz7
     def __init__(self, a=4/3):
         assert 1 <= a <= 2, "parameter out of range"
diff --git a/onn_core.py b/onn_core.py
index 881ef4e..fecb29c 100644
--- a/onn_core.py
+++ b/onn_core.py
@@ -734,7 +734,6 @@ class Model:
         offset = 0
         for node in self.ordered_nodes:
             if node.size > 0:
-                end = offset + node.size
                 inner_offset = 0
 
                 def allocate(size):