Moved model generation into own method