specify mux width by argument