Undo is now working like a champ!