Index: firmware/App/Modes/ModePostTreat.c
===================================================================
diff -u -r2df21d2472a8d79d78af7e359518acf3614accc5 -r8e7158d8231435496fcf1d5649e51babf859ccc7
--- firmware/App/Modes/ModePostTreat.c	(.../ModePostTreat.c)	(revision 2df21d2472a8d79d78af7e359518acf3614accc5)
+++ firmware/App/Modes/ModePostTreat.c	(.../ModePostTreat.c)	(revision 8e7158d8231435496fcf1d5649e51babf859ccc7)
@@ -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 ModePostTreat.c
- *
- *  @date 19-Sep-2019
- *  @author S. Nash
- *
- *  @brief Top-level state machine for the post-treatment 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 ModePostTreat.c
+*
+* @author (last) Sean Nash
+* @date (last) 13-Nov-2020
+*
+* @author (original) Dara Navaei
+* @date (original) 05-Nov-2019
+*
+***************************************************************************/
 
 #include "AlarmLamp.h"
 #include "BloodFlow.h"
@@ -26,66 +27,53 @@
     static U32 start;
 #endif
 
-    /**
-     *  @addtogroup HDPostTreatmentMode
-     *  @{
-     */
+/**
+ *  @addtogroup HDPostTreatmentMode
+ *  @{
+ */
 
 // ********** private data **********
 
 // ********** private function prototypes **********
 
-/*************************************************************************
- * @brief initPostTreatmentMode
+/*********************************************************************//**
+ * @brief
  * The initPostTreatmentMode function initializes the Post-Treatment Mode module.
- * @details
- * Inputs : none
- * Outputs : Post-Treatment Mode module initialized.
- * @param none
+ * @details Inputs: none
+ * @details Outputs: Post-Treatment Mode module initialized.
  * @return none
  *************************************************************************/
 void initPostTreatmentMode( void )
 {
 }
 
-/*************************************************************************
- * @brief transitionToPostTreatmentMode
- * The transitionToPostTreatmentMode function prepares for transition to \n
+/*********************************************************************//**
+ * @brief
+ * The transitionToPostTreatmentMode function prepares for transition to
  * post-treatment mode.
- * @details
- * Inputs : none
- * Outputs :
- * @param none
+ * @details Inputs: none
+ * @details Outputs: none
  * @return none
  *************************************************************************/
 void transitionToPostTreatmentMode( void )
 {
-    // temporary test code.  TODO - remove later
-#ifndef UF_TEST_ENABLED
-    setBloodPumpTargetFlowRate( 100, MOTOR_DIR_REVERSE, PUMP_CONTROL_MODE_OPEN_LOOP );
-    setDialInPumpTargetFlowRate( 100, MOTOR_DIR_REVERSE, PUMP_CONTROL_MODE_OPEN_LOOP );
-    setDialOutPumpTargetRate( 100, MOTOR_DIR_REVERSE, PUMP_CONTROL_MODE_OPEN_LOOP );
-#else
-#endif
 #ifdef RM46_EVAL_BOARD_TARGET
     start = getMSTimerCount();
 #endif
 }
 
-/*************************************************************************
- * @brief execPostTreatmentMode
+/*********************************************************************//**
+ * @brief
  * The execFaultMode function executes the Post-Treatment Mode state machine.
- * @details
- * Inputs : none
- * Outputs :
- * @param none
+ * @details Inputs: none
+ * @details Outputs: none
  * @return current state (sub-mode)
  *************************************************************************/
 U32 execPostTreatmentMode( void )
 {
     BOOL stop = isStopButtonPressed();
 
-#ifndef UF_TEST_ENABLED
+#ifdef DISABLE_UI_TREATMENT_WORKFLOW
     if ( TRUE == stop )
 #endif
     {