making cost decrease when deleting

1 files changed, 7 insertions, 0 deletions

diff --git a/bridge.py b/bridge.py
index ead0878..e01a866 100644
--- a/bridge.py
+++ b/bridge.py
@@ -44,9 +44,16 @@ class Bridge:
         self.joints.append(joint)
         self.add_cost(100)
         self.capacity += 500
+        
+    def joint_deleted(self):
+        self.add_cost(-100)
+        self.capacity -= 500
 
     def box_added(self):
         self.add_cost(10)
+        
+    def box_deleted(self):
+        self.add_cost(-10)
 
     def for_each_frame(self):
         self.stress = 0