now can train use csv file