Remove button needs to be added