changed class from removeMedia to removeButton