Index: leahi_dialin/td/treatment_delivery.py
===================================================================
diff -u -r01af2d84b2cb4aa87c58f10eb8af669820e71f49 -r3dc5b82f0b6f0ad88f51e709ec969b3f62af5240
--- leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 01af2d84b2cb4aa87c58f10eb8af669820e71f49)
+++ leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 3dc5b82f0b6f0ad88f51e709ec969b3f62af5240)
@@ -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
@@ -130,6 +131,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