The difference is that Atonement is playing with the unreliable narrator concept not just doing a twist ending It does work better in the book because it’s first person narrated throughout.