remove usage of m.If() in fpmax and replace with Mux