Add remove button for test