Add Model tests