Added the db code