fixed remove button function