Removed extraneous data structure and just pass the two arguments into neural_network_training.