US ends diplomatic visas for UN same-sex partners