modify train set