fixed models
Merge pull request #2 from itsmaggiemay/models

fixed models