Undo

Would be nice to have an undo functionality.