Edwin Díaz is ready to contribute in his return to the Dodger bullpen