Adding tests for Model