Added database model implementation