fixes models
Merge pull request #2 from sthulb/fixes-models

fixes models