British ambassador to Iran returns to UK