It will be great if your plugin would be able to remember the multiple focus after undoing (Ctrl + Z). Now, after applying 'undo', the focus switches to normal focus on a single line.
If you want to better understand what I mean, you can try that in Sublime Text.