Replace `mux rails` alias with more powerful function