Add undo, redo and restart button