优化Wax框架，支持函数替换