Here’s my feature suggestion. The ability to select a block of code and undo/redo only changes that occurred within that block. All edits done elsewhere in the file remain.
If you then remove any selection, all the changes that were made in that block are effectively a single new edit so you could still go back/forwards through edits in the normal manor.