Update edit.py