Adding code for the DB