modify remove button method