Add option to save model in script
Save/load model according to data set. No load on import.