Index: dialin/common/msg_ids.py
===================================================================
diff -u -raf21ad6b590a05515b33e573d1c9236f21cafa3e -r3bf31442dfb7d31dac639a2c28e43055a354b236
--- dialin/common/msg_ids.py	(.../msg_ids.py)	(revision af21ad6b590a05515b33e573d1c9236f21cafa3e)
+++ dialin/common/msg_ids.py	(.../msg_ids.py)	(revision 3bf31442dfb7d31dac639a2c28e43055a354b236)
@@ -289,3 +289,4 @@
     MSG_ID_HD_DEBUG_EVENT = 0XFFF1
     MSG_ID_DG_DEBUG_EVENT = 0XFFF2
     MSG_ID_ACK_MESSAGE_THAT_REQUIRES_ACK = 0XFFFF
+    #TODO complete as needed
\ No newline at end of file
Index: dialin/dg/flush.py
===================================================================
diff -u -recd245adee917edcb2977e109e25b8c98b3dc072 -r3bf31442dfb7d31dac639a2c28e43055a354b236
--- dialin/dg/flush.py	(.../flush.py)	(revision ecd245adee917edcb2977e109e25b8c98b3dc072)
+++ dialin/dg/flush.py	(.../flush.py)	(revision 3bf31442dfb7d31dac639a2c28e43055a354b236)
@@ -1,6 +1,5 @@
 
 import struct
-from ..utils.conversions import integer_to_bytearray
 from ..common.msg_defs import MsgIds, MsgFieldPositions
 from ..protocols.CAN import (DenaliMessage, DenaliChannels)
 from ..utils.base import _AbstractSubSystem, _publish, DialinEnum
@@ -51,7 +50,7 @@
             self.can_interface.register_receiving_publication_function(channel_id, msg_id, self._handler_flush_sync)
 
     @_publish(["flush_state", "overall_elapsed_time", "state_elapsed_time", "flush_drain_line_volume"])
-    def _handler_flush_sync(self, message):
+    def _handler_flush_sync(self, message:dict) -> None:
         """
         Handles published flush message
 
Index: tests/test_flush.py
===================================================================
diff -u -r2700dbd3c0ced061a3d8141eb00e9e3a7a3e2817 -r3bf31442dfb7d31dac639a2c28e43055a354b236
--- tests/test_flush.py	(.../test_flush.py)	(revision 2700dbd3c0ced061a3d8141eb00e9e3a7a3e2817)
+++ tests/test_flush.py	(.../test_flush.py)	(revision 3bf31442dfb7d31dac639a2c28e43055a354b236)
@@ -80,15 +80,15 @@
             if dg.dg_operation_mode == 3 or dg.dg_operation_mode == 4:
                 # If it is the first call, stop heat disinfect
                 if complete_counter == 1:
-                    dg.hd_proxy.cmd_start_stop_flush(start=False)
+                    dg.hd_proxy.cmd_start_stop_dg_flush(start=False)
                     pass
                 # Write a few more complete states to make sure the complete state items are recorded
                 elif complete_counter == 3:
                     f.close()
                     break
 
     except KeyboardInterrupt:
-        dg.hd_proxy.cmd_start_stop_flush(start=False)
+        dg.hd_proxy.cmd_start_stop_dg_flush(start=False)
         f.close()