Modify the code for the pipeline

[mmbc2008]

• Check the split between the test and the training data.
• Make sure that you do not overfit.
• Remove redundant code the was for Lizzie’s project.

[mmbc2008]

This is complete