[TECH] Introduced unit of work and modified mux to support it. <RamS>