At the end...
I honestly thought either he was gonna push her in and let her die, or that she would fall in by accident.
That would have been f'n perfect.
But the original ending works too.
I did think her dying would have been poetic justice, if you're following the "what goes around comes around" theme.
But I liked the way they ended it. Very graceful and redemptive, but somewhat bittersweet.