Add pickled model generation