Adding in the DB stuffs