Add updateData function