Adding margin back to remove button