This is a really well made puzzle platformer! I do think the ability to restart from the current stage would be good, since you are punished pretty hard for experimenting with the mechanics
I was going to devolop that idea but i ran out of time for level creation and UI design so i ended up with this less optimal version but atleast still fun version of the game. I apreatiate the feedback though!