Modified TrainNetwork2.py  add_data function to be more intuitive for the user