Algorith: Add method to train and model and save it as a picle file