Added model generation