Undo button

- Undo button works correctly
- Changed to GridLayout