ENH shuffle option for StratifiedKFold