Index: firmware/App/Modes/ModePreTreat.h
===================================================================
diff -u -r19a8bf98a7154e24c35da25225d4b55bf70ddd09 -r818d96fff963c707f5e2cb5aae7bf16f23e4b08a
--- firmware/App/Modes/ModePreTreat.h	(.../ModePreTreat.h)	(revision 19a8bf98a7154e24c35da25225d4b55bf70ddd09)
+++ firmware/App/Modes/ModePreTreat.h	(.../ModePreTreat.h)	(revision 818d96fff963c707f5e2cb5aae7bf16f23e4b08a)
@@ -8,7 +8,7 @@
 * @file    ModePreTreat.h
 *
 * @author  (last)      Dara Navaei
-* @date    (last)      09-Mar-2022
+* @date    (last)      05-Jun-2022
 *
 * @author  (original)  Dara Navaei
 * @date    (original)  05-Nov-2019
@@ -45,6 +45,7 @@
     U32 recircState;                ///< Current re-circulate state
     U32 patientConnectionState;     ///< Current patient connection state
     U32 wetSelfTestsState;          ///< CUrrent wet self-tests state
+    U32 preTreatmentRsrvrState;     ///< Current pre-treatment reservoir state
 } PRE_TREATMENT_STATE_DATA_T;
 
 // ********** public function prototypes **********
@@ -61,7 +62,8 @@
 
 void signalAlarmActionToPreTreatmentMode( ALARM_ACTION_T action );  // Execute alarm action as appropriate for pre-treatment mode
 
-BOOL getReservoirFillStatus( DG_RESERVOIR_ID_T reservoirID );
+BOOL getReservoirFillStatus( DG_RESERVOIR_ID_T reservoirID );       // Get the fill complete status for given reservoir
+HD_PRE_TREATMENT_MODE_STATE_T getPreTreatmentSubState( void );      // Get the current pre-treatment sub mode.
 
 BOOL testSetPreTreatmentModePublishIntervalOverride( U32 ms );
 BOOL testResetPreTreatmentModePublishIntervalOverride( void );