From 849ce0ea1cf391cb65bdb60bde57186995e93259 Mon Sep 17 00:00:00 2001
From: Aart van Halteren <aart.van.halteren@philips.com>
Date: Mon, 10 Jan 2022 20:45:05 +0100
Subject: [PATCH] fixed battery

---
 app/models/battery.rb | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/app/models/battery.rb b/app/models/battery.rb
index c016c95..b7ac57b 100644
--- a/app/models/battery.rb
+++ b/app/models/battery.rb
@@ -7,6 +7,10 @@ class Battery
     @battery_max_kwh = battery_capacity
   end
   
+  def reset
+    @battery_kwh = 0.0
+  end
+  
   def charge(kwh)
     return 0.0 if kwh.nil?
     if battery_kwh + kwh <= battery_max_kwh
@@ -15,7 +19,7 @@ class Battery
       return kwh
     else
       old_kwh = battery_kwh
-      battery_kwh = battery_max_kwh
+      @battery_kwh = battery_max_kwh
       #p "Battery is now at %s kwh" % battery_kwh
       return (battery_max_kwh-old_kwh)
     end
@@ -29,7 +33,7 @@ class Battery
       return kwh
     else
       old_kwh = battery_kwh
-      battery_kwh = 0.0
+      @battery_kwh = 0.0
       #p "Battery is now at %s kwh" % battery_kwh
       return old_kwh
     end