Stokes keen to play under McCullum’s white-ball regime