Created test for models
Got the tests to work