Added test for new data-model-class attribute