Adding tests for the model