Change de Train Method and created a Test Method in MainProgram.