Hey, thank you for giving my little experiment a chance! :)
The restart instead of undo button was a conscious game design choice, as I wanted to encourage the player to try a different approach to reaching the exit, if their path failed.
But I do have plans of adding temporary shadow for placed pegs, when player restarts the game. That way, you will atleast have an idea of where you had last placed your peg and replicate the path.