added class for datamodel