All functions inside train_model got its own file
Added tests to tests/testthat