This was such an amazing game! I agree that an undo button would be very helpful, and maybe dying could just undo one step, instead of resseting the whole level.
