Adding tests for model repository