diff --git a/base.py b/base.py
index 0ae520d..d7f687b 100644
--- a/base.py
+++ b/base.py
@@ -36,6 +36,7 @@ class Lo:
                         self.model[k][v] = []
                     self.model[k][v].append(lo_id)
                 self.los[lo_id] = t.copy()
+                self.los[lo_id]['weight'] = r[35]
 
     def get(self, lo_id):
         try: