Update and rename extend_func to extend_func.py