simple script for stratify. See do_stratify in impala_shell.py