fixing training data builder - csv output