I'm currently working on this new Undo feature and I'm already well underway.
I'm trying to use a single button for this feature. I was thinking of doing it this way: a single button press takes you back one block, while a longer press restarts the level. What do you think?