Model generator
Add clasess