Index: firmware/App/Controllers/Ultrafiltration.h
===================================================================
diff -u -rab9df65fbbb4f8387a5f2c0e7aa26bd234843150 -r4237cbced773b985ff69a8c2af8287dbd5457f24
--- firmware/App/Controllers/Ultrafiltration.h	(.../Ultrafiltration.h)	(revision ab9df65fbbb4f8387a5f2c0e7aa26bd234843150)
+++ firmware/App/Controllers/Ultrafiltration.h	(.../Ultrafiltration.h)	(revision 4237cbced773b985ff69a8c2af8287dbd5457f24)
@@ -1,17 +1,17 @@
 /**************************************************************************
 *
-* Copyright (c) 2024-2025 Diality Inc. - All Rights Reserved.
+* Copyright (c) 2025-2026 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    Ultrafiltration.h
 *
-* @author  (last)      Vinayakam Mani
-* @date    (last)      23-Jul-2025
+* @author  (last)      Jashwant Gantyada
+* @date    (last)      30-Jun-2026
 *
 * @author  (original)  Vinayakam Mani
-* @date    (original)  23-Jul-2025
+* @date    (original)  28-Jul-2025
 *
 ***************************************************************************/
 
@@ -34,7 +34,6 @@
 /// ultrafiltration data structure
 typedef struct
 {
-    U32 ufExecState;                                                    ///< Ultrafiltration execution state
     F32 ufRate;                                                         ///< Ultrafiltration rate from TD
     F32 compUFrate;                                                     ///< Compensated UF rate
     U32 isUFRequested;                                                  ///< Ultrafiltration run or puase request
@@ -44,8 +43,7 @@
 
 void initUltrafiltration( void );                                        // Initialize ultrafiltration unit
 void transitionToUltrafiltration( void );                                // Prepares for transition to ultrafiltration execution
-U32 execUFControl( void );                                               // Execute the ultrafiltration state machine
-UF_EXEC_STATE_T getCurrentUFExecState( void );                           // Get the current state of the balancing chamber execution
+void handleUFControl( void );                                            // Handle ultrafiltration control
 void signalUFRateUpdate( void );                                         // Update UF rate when there is a change in rate
 
 BOOL testDDUFDataPublishIntervalOverride( MESSAGE_T *message );          // To override the UF data publish interval