save_model function

Saving the machine learning model on the disk