Adding Tests for the Model