I like it but maybe add an undo button?