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