Index: firmware/App/Modes/ModeTreatment.h
===================================================================
diff -u -r933a18d740285e70be9d00696ed0f5a5381bc8e4 -rc282822f36836a8127f447c8ac5b8a50e851be63
--- firmware/App/Modes/ModeTreatment.h	(.../ModeTreatment.h)	(revision 933a18d740285e70be9d00696ed0f5a5381bc8e4)
+++ firmware/App/Modes/ModeTreatment.h	(.../ModeTreatment.h)	(revision c282822f36836a8127f447c8ac5b8a50e851be63)
@@ -46,6 +46,7 @@
     U32 treatmentSubMode;
     U32 uFState;
     U32 salineBolusState;
+    U32 heparinState;
 } TREATMENT_STATE_DATA_T;
 
 /// Payload record structure for an ultrafiltration volume change confirmation message.
@@ -77,14 +78,14 @@
 
 // ********** private function prototypes **********
 
-void initTreatmentMode( void );             // initialize this module
-void transitionToTreatmentMode( void );     // prepares for transition to treatment mode
-U32  execTreatmentMode( void );             // execute the treatment mode state machine (call from OperationModes)
+void initTreatmentMode( void );                                     // Initialize this module
+void transitionToTreatmentMode( void );                             // Prepares for transition to treatment mode
+U32  execTreatmentMode( void );                                     // Execute the treatment mode state machine (call from OperationModes)
+void signalAlarmActionToTreatmentMode( ALARM_ACTION_T action );     // Execute alarm action as appropriate for Treatment mode
 
-TREATMENT_STATE_T getTreatmentState( void );
+TREATMENT_STATE_T getTreatmentState( void );                        // Determine the current treatment sub-mode (state)
 
-BOOL userRequestEndTreatment( void );
-void broadcastTreatmentTimeAndState( void );
+void broadcastTreatmentTimeAndState( void );                        // Broadcast the times and states of this treatment
 
 BOOL verifyTreatmentDurationSettingChange( U32 treatmentTime );
 BOOL verifyUFSettingsChange( F32 uFVolume );