Add model generator

Sorry that I missed this :p