Ben Stokes feels his time away from cricket to reflect on his mental health will aid his captaincy skills