What would have made the ending slightly better (spoiler, natch)
That was a great twist ending; I didn't really see it happening. Even when Eleanor said it was the bad place, I thought that they had made a model of the good place in the bad place to trick them and that only the good people would leave at the end.
So at the end of the episode, their memories were erased, things were changed and they were given another run.
What I think would have made it slightly better is if they were told that this was the 1,200th time that they had been through the scenario - that they had already been in the bad place for, say, a millenia, and that they had had their minds erased many times.