Add support for function creation