Add model generator