Ben Stokes returns to the England set-up after his court case as head coach Trevor Bayliss encourages the all-rounder to make a public apology.