Add model tests
Merge pull request #10 from chriswoodrich/model_tests

Add model tests