Stokes given England contract despite mental health break