Update extend_func

修正一些bug