It would be great to be able to select a chunk of text (via the VISUAL modes) and for undotree to only undo changes in that selected part of the file.
For example this would probably work well with functions/classes/etc to only undo changes that were made to that specific structure.