TrainModels implemented

with parallelization (to do)