Save trained models

Training models takes a lot of time in the pipeline. This could be avoided by saving models using the pickle module.