Add redo code