Added data model to API