England hit back as Stokes' retirement fires four-wicket session