Edit Distance Algorithm Python