Started adding 'undo' button. INCOMPLETE