Add remove button

Remove functionality