Added progression and model saving to script