working on the remove button