Add a setFlatFunc function.