ready to roll -- main_training_generic should be tested, plotting todo