US judge limits Trump's ability to withhold school funds over DEI