made the complex_training_file thingy more flexible