Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -r549119eae64732f124d22df66de4fc88c56193c0 -r113f1633a8d944d0bdd644c9b866ed58822a14b7
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 549119eae64732f124d22df66de4fc88c56193c0)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 113f1633a8d944d0bdd644c9b866ed58822a14b7)
@@ -193,7 +193,9 @@
     treatmentStartTimeStamp = getRTCTimestamp();
     treatmentEndTimeStamp = 0;
 
-    resetBloodPumpRotorCount();  // Starting up or starting new treatment - reset rotor count to indicate new cartridge (no wear).
+    initBloodFlow();
+    initDialInFlow();
+    initDialOutFlow();
 }
 
 /*********************************************************************//**
@@ -362,6 +364,19 @@
 
 /*********************************************************************//**
  * @brief
+ * The getCurrentUFSetRate function returns the currently set target
+ * ultrafiltration rate.
+ * @details Inputs: presUFRate
+ * @details Outputs: none
+ * @return presUFRate
+ *************************************************************************/
+F32 getCurrentUFSetRate( void )
+{
+    return presUFRate;
+}
+
+/*********************************************************************//**
+ * @brief
  * The getTreatmentAvgBloodFlowRate function returns the average blood flow
  * rate collected through out the treatment.
  * @details Inputs: treatmentTimeMS, treatmentBloodFlowRateTotal_mL_min