added action on the button Remove and Restore