Index: firmware/App/Modes/ModeDrain.h
===================================================================
diff -u -rc48a99d2d1c852adcc986253b6c420a90dab7bfe -rf5b02f03b6695c0c76fd8a4d902a13114e1a8aca
--- firmware/App/Modes/ModeDrain.h	(.../ModeDrain.h)	(revision c48a99d2d1c852adcc986253b6c420a90dab7bfe)
+++ firmware/App/Modes/ModeDrain.h	(.../ModeDrain.h)	(revision f5b02f03b6695c0c76fd8a4d902a13114e1a8aca)
@@ -18,6 +18,7 @@
 #define __MODE_DRAIN_H__
 
 #include "DGCommon.h"
+#include "DGModes.h"
 
 /**
  *  @defgroup DrainMode DrainMode
@@ -28,12 +29,16 @@
  *  @{
  */
 
+// ********** public definitions **********
+
 // ********** private function prototypes **********
 
 void initDrainMode( void );         // initialize this module
-void transitionToDrainMode( void );    // prepares for transition to operating parameters mode
-void execDrainMode( void );         // execute the drain mode state machine
+void 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.
+
 /**@}*/
 
 #endif