Break up the train function to individual functions based on em_bright, mass_gap, or ssm

Currently, the train function decides a setting based on command line arguments and then does the workflow based on conditional statements. This latter piece should be broken up into three separate functions that will be decided based on the settings.