worked on remove button