Index: dialin/hd/treatment.py
===================================================================
diff -u -r8c1ae1a592e02562ad9bded9ea2bd199fa80f4bf -r12f3865099530cf08e9197e501a9507250729e77
--- dialin/hd/treatment.py	(.../treatment.py)	(revision 8c1ae1a592e02562ad9bded9ea2bd199fa80f4bf)
+++ dialin/hd/treatment.py	(.../treatment.py)	(revision 12f3865099530cf08e9197e501a9507250729e77)
@@ -128,6 +128,7 @@
         self.blood_prime_state = 0
         self.treatment_end_state = 0
         self.treatment_stop_state = 0
+        self.dialysis_state = 0
         # saline bolus status
         self.saline_bolus_max_vol = 0
         self.saline_bolus_cum_vol = 0.0
@@ -197,7 +198,6 @@
         # blood prime status
         self.blood_prime_tgt_vol = 0.0
         self.blood_prime_cum_vol = 0.0
-        self.blood_prime_ind_cum_vol = 0.0
         # rinseback status
         self.rinseback_tgt_vol = 0.0
         self.rinseback_cum_vol = 0.0
@@ -207,7 +207,11 @@
         # re-circulation status
         self.recirc_timeout_secs = 0
         self.recirc_countdown_secs = 0
+        # treatment stop status
+        self.treatment_stop_timeout_secs = 0
+        self.treatment_stop_timeout_coundown_secs = 0
 
+
     def get_treatment_time_prescribed(self) -> int:
         """
         Gets the prescribed treatment time