Replaced the remove button with an icon