Add output mux generation