add train/test/slpit