US withdraws diplomat after India expulsion demand