remove button is now working.