Index: tst_end_treatment/test.py
===================================================================
diff -u -rfc2bd008f7fbaa68dc4b29f9219c506a2830cce1 -reca0afd252f1c84a6711492854c08ec03266ca9e
--- tst_end_treatment/test.py	(.../test.py)	(revision fc2bd008f7fbaa68dc4b29f9219c506a2830cce1)
+++ tst_end_treatment/test.py	(.../test.py)	(revision eca0afd252f1c84a6711492854c08ec03266ca9e)
@@ -25,11 +25,10 @@
 
 hd_simulator = HDSimulator()
        
-def verification_of_end_treatment_state():    
+def verification_of_end_treatment_complete_state():    
     """
-         Method to verify End treatment states
+         Method to verify End treatment Treatment Complete state screens
     """
-    test.startSection("verifying End treatment state screens")
     test.startSection("verifying End treatment Treatment Complete state screens")
     hd_simulator.cmd_set_treatment_states_data(sub_mode = TreatmentStates.TREATMENT_END_STATE.value , uf_state = TreatmentStates.TREATMENT_START_STATE.value, 
                                                saline_state = TreatmentStates.TREATMENT_START_STATE.value, heparin_state = TreatmentStates.TREATMENT_START_STATE.value,rinseback_state= TreatmentStates.TREATMENT_START_STATE.value, recirculate_state = TreatmentStates.TREATMENT_START_STATE.value, 
@@ -40,22 +39,27 @@
     test.verify(waitForObjectExists(names.o_start_rinseback_button_text).enabled, " Start Rinseback button must be active")
     mouseClick(waitForObject(names.o_start_rinseback_button_text))
     test.endSection()
-    
+
+def verification_of_end_treatment_paused_state():    
+    """
+         Method to verify End treatment Treatment Paused state screens
+    """    
     test.startSection("verifying End treatment Treatment Paused state screens")
     hd_simulator.cmd_set_treatment_states_data(sub_mode = TreatmentStates.TREATMENT_END_STATE.value , uf_state = TreatmentStates.TREATMENT_START_STATE.value, 
                                                saline_state = TreatmentStates.TREATMENT_START_STATE.value, heparin_state = TreatmentStates.TREATMENT_START_STATE.value,rinseback_state= TreatmentStates.TREATMENT_START_STATE.value, recirculate_state = TreatmentStates.TREATMENT_START_STATE.value, 
                                                blood_prime_state = TreatmentStates.TREATMENT_START_STATE.value,treatment_end_state = TreatmentEndStates.TREATMENT_END_PAUSED_STATE.value, treatment_stop_state = TreatmentStates.TREATMENT_START_STATE.value, dialysis_state = TreatmentStates.TREATMENT_START_STATE.value)
     
     test.compare(str(waitForObject(names.o_end_treatment_complete_paused_state).text),config.END_TREATMENT_COMPLETE_PAUSED_STATE_TEXT, "Treatment text must be {}".format(config.END_TREATMENT_COMPLETE_PAUSED_STATE_TEXT))
     test.endSection()
-    test.endSection()
+
     
     
 
 def main():
     
     utils.tstStart(__file__)
     startApplication(config.AUT_NAME) 
-    verification_of_end_treatment_state()
+    verification_of_end_treatment_complete_state()
+    verification_of_end_treatment_paused_state()