Updated the database model