Adding pickle model generation script