Added Data model class