Add remove button
Achieve remove button