Update remove button functionnality