Index: leahi_dialin/td/treatment_delivery.py
===================================================================
diff -u -r95205b22fbcf8ce097fd7dba0a34156e5f9008a5 -rf398cb62d5ddc8385fd6b7f38b35e035964c9570
--- leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 95205b22fbcf8ce097fd7dba0a34156e5f9008a5)
+++ leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision f398cb62d5ddc8385fd6b7f38b35e035964c9570)
@@ -30,6 +30,8 @@
 from .modules.dry_self_test import TDDrySelfTest
 from .modules.ejector import TDEjector
 from .modules.events import TDEvents
+from .modules.post_treatment import TDPostTreatment
+from .modules.pre_treatment import TDPreTreatment
 from .modules.pressure_sensors import TDPressureSensors
 from .modules.records import TDRecords
 from .modules.switches import TDSwitches
@@ -137,6 +139,8 @@
         self.dry_self_test = TDDrySelfTest(self.can_interface, self.logger)                 #: The Dry Self Test module
         self.ejector = TDEjector(self.can_interface, self.logger)                           #: The Ejector module
         self.events = TDEvents(self.can_interface, self.logger)                             #: The Events module
+        self.post_treatment = TDPostTreatment(self.can_interface, self.logger)              #: The Post-Treatment module
+        self.pre_treatment = TDPreTreatment(self.can_interface, self.logger)                #: The Pre-Treatment module
         self.pressure_sensors = TDPressureSensors(self.can_interface, self.logger)          #: The Pressure Sensors module
         self.records = TDRecords(self.can_interface, self.logger)                           #: The Records module
         self.switches = TDSwitches(self.can_interface, self.logger)                         #: The Switches module