With the new apple pencil pro, integrate the option to squeeze to erase and then when you stop, it goes back to your previous tool. I believe this is a new feature in notability.