Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -rd91a24c730aeb5cd7e3eba9ef4eca78e442911f8 -rd9d085cdca67200ecddbdfbc75c489e704b23081
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision d91a24c730aeb5cd7e3eba9ef4eca78e442911f8)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision d9d085cdca67200ecddbdfbc75c489e704b23081)
@@ -15,6 +15,7 @@
 *
 ***************************************************************************/
 
+#include "AirTrap.h"
 #include "AlarmLamp.h"
 #include "BloodFlow.h"
 #include "Buttons.h"
@@ -205,6 +206,8 @@
     {
         requestNewOperationMode( MODE_POST );
     }
+    broadcastTreatmentTimeAndState();
+    broadcastTreatmentSettingsRanges();
 #else
     // treatment mode state machine
     switch ( currentTreatmentState )
@@ -251,6 +254,7 @@
 
         case TREATMENT_END_STATE:
             // TODO - implement
+            endAirTrapControl();  // TODO - move to appropriate place
             requestNewOperationMode( MODE_POST );  // TODO - test code - remove later
             break;