correct train/test exampels