From an obsession with fitness to fixing bad habits, Stokes' return to form with the ball is a boon for England