Add an option for custom functions