Added test for data model