- add more Mux optimization cases