Adding more tests for model