train set generation changes