Add some model tests
Added tests for models