Adding tests for model