Index: firmware/App/Modes/ModeDrain.h
===================================================================
diff -u -r4efb2d1786c717ca6cf4b1bc992cec8e16820bff -r94a190522ce398399c7b93c59f788d7666ec0060
--- firmware/App/Modes/ModeDrain.h	(.../ModeDrain.h)	(revision 4efb2d1786c717ca6cf4b1bc992cec8e16820bff)
+++ firmware/App/Modes/ModeDrain.h	(.../ModeDrain.h)	(revision 94a190522ce398399c7b93c59f788d7666ec0060)
@@ -1,18 +1,19 @@
-/**********************************************************************//**
- *
- *  Copyright (c) 2019-2020 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
- *
- *  @date 19-Sep-2019
- *  @author S. Nash
- *
- *  @brief Header file for Drain Mode.
- *
- **************************************************************************/
+/**************************************************************************
+*
+* 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
+*
+* @author  (last)      Dara Navaei
+* @date    (last)      24-Oct-2021
+*
+* @author  (original)  Leonardo Baloa
+* @date    (original)  20-Dec-2019
+*
+***************************************************************************/
 
 #ifndef __MODE_DRAIN_H__
 #define __MODE_DRAIN_H__
@@ -22,19 +23,18 @@
 
 /**
  *  @defgroup DGDrainMode DGDrainMode
- *  @brief Drain mode module.
- *  Manages drain mode functions via a state machine.
+ *  @brief Drain mode module. Manages drain mode functions via a state machine.
  *
  *  @addtogroup DGDrainMode
  *  @{
  */
 
 // ********** public definitions **********
 
-// ********** private function prototypes **********
+// ********** 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.