Adding tests for model manager