Add support for calling functions with functions as params