Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -r5e01e06bf4c6b05665ef096ea873d09b7bc904d3 -r58afeb5a318882fa028894b51aefba526b2fb1f0
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 5e01e06bf4c6b05665ef096ea873d09b7bc904d3)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 58afeb5a318882fa028894b51aefba526b2fb1f0)
@@ -192,7 +192,9 @@
     treatmentStartTimeStamp = getRTCTimestamp();
     treatmentEndTimeStamp = 0;
 
-    resetBloodPumpRotorCount();  // Starting up or starting new treatment - reset rotor count to indicate new cartridge (no wear).
+    initBloodFlow();
+    initDialInFlow();
+    initDialOutFlow();
 }
 
 /*********************************************************************//**
@@ -361,6 +363,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