US officials have been forced to defend the pace of investigation into Volkswagen's 'dieselgate' after being put on the spot by a judge.