remove button is now working