Updates styling of undo button