Stokes is aiming to return in time for the start of the English Test summer after tearing his hamstring in New Zealand