minor
switching between train and test