Please implement the undo logic

Great, works perfectly, but if you can please make it possible that when I click again it will undo the layout change, because I need to update the page to undo.