More changes in Undo button