Update extend_func.py