Update edit_distance.py