Update edit_handler.py