Change add parameter to support function and function decl