Add simple model generator test
Add model generator and test