Update 01_function.py