Index: firmware/App/Modes/ModeHeatDisinfect.h
===================================================================
diff -u -rcea079b61dbd17b2ddaec99b1124248147d14e72 -rfd84d51bd0177b49578859572d24f68571280ea4
--- firmware/App/Modes/ModeHeatDisinfect.h	(.../ModeHeatDisinfect.h)	(revision cea079b61dbd17b2ddaec99b1124248147d14e72)
+++ firmware/App/Modes/ModeHeatDisinfect.h	(.../ModeHeatDisinfect.h)	(revision fd84d51bd0177b49578859572d24f68571280ea4)
@@ -1,17 +1,17 @@
 /**************************************************************************
 *
-* Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
+* Copyright (c) 2020-2023 Diality Inc. - All Rights Reserved.
 *
 * THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
 * WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
 *
-* @file ModeHeatDisinfect.h
+* @file    ModeHeatDisinfect.h
 *
-* @author (last) Quang Nguyen
-* @date (last) 24-Aug-2020
+* @author  (last)      Michael Garthwaite
+* @date    (last)      29-Aug-2023
 *
-* @author (original) Sean
-* @date (original) 20-Apr-2020
+* @author  (original)  Sean
+* @date    (original)  20-Apr-2020
 *
 ***************************************************************************/
 
@@ -35,19 +35,30 @@
 typedef struct
 {
     U32 heatDisinfectTargetTime;        ///< Target time to heat disinfect.
-    U32 heatDisinfectCountdownTime;     ///< Heat disinfect count down time.
+    U32 ro77CountdownTimeS;             ///< Heat disinfect RO 77 count down time in seconds.
+    U32 ro82CountdownTimeS;             ///< Heat disinfect RO 82 count down time in seconds.
+    U32 r77CountdownTimeS;              ///< Heat disinfect reservoir 77 count down time in seconds.
+    U32 r82CountdownTimeS;              ///< Heat disinfect reservoir 82 count down time in seconds.
 } MODE_HEAT_DISINFECT_UI_DATA_T;
 
 // ********** public function prototypes **********
 
 void initHeatDisinfectMode( void );                                 // Initialize this module
-void transitionToHeatDisinfectMode( void );                         // Prepares for transition to heat disinfect mode
+U32 transitionToHeatDisinfectMode( void );                          // Prepares for transition to heat disinfect mode
 U32  execHeatDisinfectMode( void );                                 // Execute the heat disinfect mode state machine (call from OperationModes)
 
 DG_HEAT_DISINFECT_STATE_T getCurrentHeatDisinfectState( void );     // get the current state of the heat disinfect mode
 
 BOOL stopDGHeatDisinfect( void );                                   // Stop DG heat disinfect
+BOOL testSetDG77CStateTimerOverride( U32 timer );                   // set override timer for disinfection at 77C
+BOOL testResetDG77CStateTimerOverride( void );                      // reset override timer for disinfection at 77C
+BOOL testSetDG82CStateTimerOverride( U32 timer );                   // set override timer for disinfection at 82C
+BOOL testResetDG82CStateTimerOverride( void );                      // reset override timer for disinfection at 77C
 
+#ifndef _RELEASE_
+void setHeatNelsonSupportMode( NELSON_SUPPORT_T support );
+#endif
+
 /**@}*/
 
 #endif