Update edit_functions.py