Index: leahi_dialin/td/treatment_delivery.py
===================================================================
diff -u -reca9e15bc984af2b7f59eb38f3836ca354ee1251 -r43f69b19f03145572d88f5f3c0cc739503c6d647
--- leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision eca9e15bc984af2b7f59eb38f3836ca354ee1251)
+++ leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 43f69b19f03145572d88f5f3c0cc739503c6d647)
@@ -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.blood_pressure import TDBloodPressure
 from .modules.bubble_detector import TDBubbleDetector
@@ -127,6 +128,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.blood_pressure = TDBloodPressure(self.can_interface, self.logger)              #: The Blood Pressure module
         self.bubbles = TDBubbleDetector(self.can_interface, self.logger)                    #: The Bubble Detector module