Adding model code coverage