first commit. Build the basic infrastructure for the training