Judge back with the Patriots