Index: leahi_dialin/td/treatment_delivery.py
===================================================================
diff -u -r88eea899fa8d03596f944505285bf3049e64312b -r12f6d9b3476d413d861dc9589635ec05bf8e6f9d
--- leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 88eea899fa8d03596f944505285bf3049e64312b)
+++ leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 12f6d9b3476d413d861dc9589635ec05bf8e6f9d)
@@ -22,6 +22,7 @@
 from .modules.air_pump import TDAirPump
 from .modules.air_trap import TDAirTrap
 from .modules.alarms import TDAlarms
+from .modules.battery import TDBattery
 from .modules.blood_flow import TDBloodFlow
 from .modules.bubble_detector import TDBubbleDetector
 from .modules.buttons import TDButtons
@@ -128,6 +129,7 @@
         self.air_pump = TDAirPump(self.can_interface, self.logger)                          #: The Air Pump module
         self.air_trap = TDAirTrap(self.can_interface, self.logger)                          #: The Air Trap module
         self.alarms = TDAlarms(self.can_interface, self.logger)                             #: The Alarms module
+        self.battery = TDBattery(self.can_interface, self.logger)                           #: The Battery module
         self.blood_flow = TDBloodFlow(self.can_interface, self.logger)                      #: The Blood Flow module
         self.bubbles = TDBubbleDetector(self.can_interface, self.logger)                    #: The Bubble Detector module
         self.buttons = TDButtons(self.can_interface, self.logger)                           #: The Buttons module