The government faced pushback when its lawyer tried to convince the court to let it broadly enforce the Trump's birthright citizenship rules.