Added model generator