updating the layout to have the remove button