Add updateDB logic