Top UK diplomat visits US to reassure over Brexit bill