Undo button
Initial stuff for undo functionality