added ability to pick number of examples to training with and writing more info to csv