Add database init function