Changes to FunctionAux to make it work with higher order elemental variables