Index: firmware/App/Modes/ModeDrain.h
===================================================================
diff -u -r24dd186948c13ae8e1ff88c14cc4f478e739ee0b -rdcd360fb4dc37db2dcbeb7fb14fb327fe68235f4
--- firmware/App/Modes/ModeDrain.h	(.../ModeDrain.h)	(revision 24dd186948c13ae8e1ff88c14cc4f478e739ee0b)
+++ firmware/App/Modes/ModeDrain.h	(.../ModeDrain.h)	(revision dcd360fb4dc37db2dcbeb7fb14fb327fe68235f4)
@@ -1,17 +1,17 @@
 /**************************************************************************
 *
-* Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
+* Copyright (c) 2019-2021 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 ModeDrain.h
+* @file    ModeDrain.h
 *
-* @author (last) Quang Nguyen
-* @date (last) 24-Aug-2020
+* @author  (last)      Dara Navaei
+* @date    (last)      24-Oct-2021
 *
-* @author (original) Leonardo Baloa
-* @date (original) 20-Dec-2019
+* @author  (original)  Leonardo Baloa
+* @date    (original)  20-Dec-2019
 *
 ***************************************************************************/
 
@@ -34,7 +34,7 @@
 // ********** public function prototypes **********
 
 void initDrainMode( void );         // initialize this module
-void transitionToDrainMode( void ); // prepares for transition to operating parameters mode
+U32 transitionToDrainMode( void ); // prepares for transition to operating parameters mode
 U32 execDrainMode( void );          // execute the drain mode state machine
 
 DG_DRAIN_STATE_T getCurrentDrainState( void );  // get the current state of the drain mode.